const { generateTestData, cleanTestData } = require('./seedUtils');
const args = process.argv.slice(2);

if (args.includes('--generate')) {
    const count = parseInt(args[args.indexOf('--generate') + 1]) || 20;
    generateTestData(count);
} else if (args.includes('--clean')) {
    cleanTestData(args.includes('--all'));
} else {
    console.log('可用命令：');
    console.log('--generate [数量] 创建测试数据（默认20条）');
    console.log('--clean [--all]    清理测试数据（加--all清理全部）');
}